-
Notifications
You must be signed in to change notification settings - Fork 790
[Joint Matrix] Enable different accumulator and output types in spirv. Add tests to cover bfloat16 and half floating-point sizes. #17502
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[Joint Matrix] Enable different accumulator and output types in spirv. Add tests to cover bfloat16 and half floating-point sizes. #17502
Conversation
f5e46ba
to
3a5de5a
Compare
3a5de5a
to
c5be2c5
Compare
c5be2c5
to
29c35ac
Compare
29c35ac
to
8116625
Compare
8116625
to
fc7eaf7
Compare
409d5c2
to
a82ba68
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
@YuriPlyakhin, can you approve the requested changes if you are happy with the changes made? |
@intel/llvm-reviewers-runtime, please help review the non matrix parts |
@intel/llvm-gatekeepers, this PR is ready to be merged. |
Need @YuriPlyakhin to unblock. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM. With quick final review I don't see any problems, but I'm glad Dounia reviewed it as well, as after multiple review rounds my attention to detail may be less sharp.
@ggojska you need to make your e-mail public on Github, I think. |
@aelovikov-intel it should be public now |
Updated matrix_compare and matrix_multiply_ref functions to better match bfloat16 calculations on device.